package com.quick.common.core.domain;

import org.springframework.http.HttpStatus;

import java.util.HashMap;

/**
 * 响应报文
 *
 * @Author: Administrator
 * @Date: 2023/7/17
 */
public class ApiResponse extends HashMap<String, Object> {
    /**
     * 状态码
     */
    public static final String CODE = "code";
    /**
     * 返回内容
     */
    public static final String MSG = "msg";
    /**
     * 数据对象
     */
    public static final String DATA = "data";
    /**
     * 数据数量
     */
    public static final String TOTAL = "total";

    private static final long serialVersionUID = 1L;


    /**
     * 初始化一个新创建的 ApiResponse 对象
     *
     * @param code 状态码
     * @param msg  返回内容
     */
    public ApiResponse(int code, String msg) {
        super.put(CODE, code);
        super.put(MSG, msg);
    }

    /**
     * 初始化一个新创建的 ApiResponse 对象
     *
     * @param code 状态码
     * @param msg  返回内容
     * @param data 数据对象
     */
    public ApiResponse(int code, String msg, Object data) {
        super.put(CODE, code);
        super.put(MSG, msg);
        super.put(DATA, data);
    }

    /**
     * 返回成功消息
     *
     * @return 成功消息
     */
    public static ApiResponse success() {
        return success(null);
    }

    /**
     * 返回成功数据
     *
     * @return 成功消息
     */
    public static ApiResponse success(Object data) {
        return success("操作成功！", data);
    }

    /**
     * 返回成功数据
     *
     * @return 成功消息
     */
    public static ApiResponse success(int row) {
        return row > 0 ? success() : error();
    }


    /**
     * 返回成功消息
     *
     * @param msg  返回内容
     * @param data 数据对象
     * @return 成功消息
     */
    public static ApiResponse success(String msg, Object data) {
        return new ApiResponse(HttpStatus.OK.value(), msg, data);
    }

    /**
     * 返回错误消息
     *
     * @return 警告消息
     */
    public static ApiResponse error() {
        return error("操作失败！");
    }

    /**
     * 返回错误消息
     *
     * @param msg 返回内容
     * @return 警告消息
     */
    public static ApiResponse error(String msg) {
        return error(msg, null);
    }

    /**
     * 返回错误消息
     *
     * @param msg  返回内容
     * @param data 数据对象
     * @return 警告消息
     */
    public static ApiResponse error(String msg, Object data) {
        return new ApiResponse(HttpStatus.INTERNAL_SERVER_ERROR.value(), msg, data);
    }

    /**
     * 返回错误消息
     *
     * @param code 状态码
     * @param msg  返回内容
     * @return 警告消息
     */
    public static ApiResponse error(int code, String msg) {
        return new ApiResponse(code, msg, null);
    }

    /**
     * 方便链式调用
     *
     * @param key   键
     * @param value 值
     * @return 数据对象
     */
    @Override
    public ApiResponse put(String key, Object value) {
        super.put(key, value);
        return this;
    }
}
